FROM tensorflow/tensorflow:1.12.0-gpu

WORKDIR /root/

ENV LANG=C.UTF-8

RUN apt-get update && apt-get install -y wget vim libopencv-dev python-numpy python-scipy python-matplotlib

RUN pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install keras -i https://pypi.tuna.tsinghua.edu.cn/simple 

CMD exec /bin/bash -c "service ssh start;trap : TERM INT; sleep infinity & wait"
